package thread;

import java.util.PriorityQueue;

class MyTimerTask implements Comparable<MyTimerTask> {
    private Runnable task;
    // 这个地方为了方便不和“当前时间对比”，确认这个任务是否要执行，保存绝对的时间戳，更方便的
    private long time;

    public MyTimerTask(Runnable task, long delay) {
        this.task = task;
        this.time = System.currentTimeMillis() + delay;

    }

    public Runnable getTask() {
        return task;
    }

    public long getTime() {
        return time;
    }

    @Override
    public int compareTo(MyTimerTask o) {
        // 小堆的效果
        return (int) (this.time - o.time);
    }
}

// 模拟实现定时器
// 定时器要能够实现同时管理多个任务
// 有一个的数据结构来组织这个任务
// ArrayList不太合适
// 更好的选择是优先级队列
class MyTimer {
    private PriorityQueue<MyTimerTask> queue = new PriorityQueue<>();
    private Object locker = new Object();

    public MyTimer() {
        // 创建线程，检测任务是否到时间了，以及去执行这个任务
        Thread t = new Thread(() -> {
            // 循环的从队列中取出元素，然后来判断这个任务时间是否到达
            // 如果到达就执行任务，并且出队列
            // 如果没到达时间，就啥也不做
            while (true) {
                try {
                    synchronized (locker) {
                        if (queue.isEmpty()) {
                            locker.wait();
                        }
                        // 取出队首元素
                        MyTimerTask task = queue.peek();
                        // 记录下当前时刻的时间戳
                        long curTime = System.currentTimeMillis();
                        if (curTime < task.getTime()) {
                            locker.wait(task.getTime()-curTime);
                        } else {
                            // 时间到了，执行任务
                            task.getTask().run();
                            queue.poll();
                        }
                    }
                } catch (InterruptedException e) {
                    // TODO: handle exception
                    e.printStackTrace();
                }

            }
        });
        t.start();

    }

    public void schedule(Runnable runnable, long delay) {
        synchronized (locker) {
            queue.offer(new MyTimerTask(runnable, delay));
            locker.notify();
        }
    }
}

public class demo37 {
    public static void main(String[] args) {
        MyTimer myTimer = new MyTimer();
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("定时任务 3000");
            }
        }, 3000);

        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("定时任务 2000");
            }
        }, 2000);
        myTimer.schedule(new Runnable() {
            @Override
            public void run() {
                System.out.println("定时任务 1000");
            }
        }, 1000);
    }
}
